// Copyright (c) 2011 SameGoal LLC.
// All Rights Reserved.
// Author: Andy Hochhaus <ahochhaus@samegoal.com>

// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#ifndef NET_HTTP_SERVER_COMET_REQUEST_HELPER_H_
#define NET_HTTP_SERVER_COMET_REQUEST_HELPER_H_

#include <event2/buffer.h>
#include <event2/bufferevent.h>
#include <event2/bufferevent_struct.h>
#include <event2/event.h>
#include <event2/event_struct.h>
#include <fcntl.h>
#include <gflags/gflags.h>
#include <glog/logging.h>

#include <map>
#include <string>

#include "base/basictypes.h"
#include "net/http/server/event/request.h"

namespace sg {
namespace net {
namespace http {
namespace server {
namespace comet {

namespace es = ::sg::net::http::server::event;

class RequestHelper {
 public:
  enum PaddingType { NONE, LENGTH, SCRIPT };

  explicit RequestHelper(es::Request *req);

  bool StartChunk(const PaddingType padding);
  bool SendChunk(const std::string &body, const PaddingType padding);
  bool EndChunk(const PaddingType padding);
  bool Send(const es::HttpStatus status_code,
            const std::string &body,
            const PaddingType padding);

 private:
  bool FillPaddedBuffer(const std::string &body,
                        const PaddingType padding,
                        std::string *buf);

  es::Request *req_;

  DISALLOW_COPY_AND_ASSIGN(RequestHelper);
};

}  // namespace comet
}  // namespace server
}  // namespace http
}  // namespace net
}  // namespace sg

#endif  // NET_HTTP_SERVER_COMET_REQUEST_HELPER_H_
